package io.renren.utils;


import io.renren.common.QpayConstants;
import org.jdom2.Document;
import org.jdom2.Element;
import org.jdom2.JDOMException;
import org.jdom2.input.SAXBuilder;

import javax.servlet.http.HttpServletRequest;
import java.io.*;
import java.net.ConnectException;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.*;


/**
 * @Auther: gxd
 * @Date: 2018/6/23 16:14
 * @Description:微信支付工具类
 */
public class WXPayCommonUtil {

    //随机字符串生成
    public static String getRandomString(int length) { //length表示生成字符串的长度
        String base = "abcdefghijklmnopqrstuvwxyz0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ";
        Random random = new Random();
        StringBuffer sb = new StringBuffer();
        for (int i = 0; i < length; i++) {
            int number = random.nextInt(base.length());
            sb.append(base.charAt(number));
        }
        return sb.toString();
    }

    //请求xml组装
    public static String getRequestXml(SortedMap<String, Object> parameters) {
        StringBuffer sb = new StringBuffer();
        sb.append("<xml>");
        Set es = parameters.entrySet();
        Iterator it = es.iterator();
        while (it.hasNext()) {
            Map.Entry entry = (Map.Entry) it.next();
            String key = (String) entry.getKey();
            String value = (String) entry.getValue();
            if ("attach".equalsIgnoreCase(key) || "body".equalsIgnoreCase(key) || "sign".equalsIgnoreCase(key)) {
                sb.append("<" + key + ">" + "<![CDATA[" + value + "]]></" + key + ">");
            } else {
                sb.append("<" + key + ">" + value + "</" + key + ">");
            }
        }
        sb.append("</xml>");
        return sb.toString();
    }

    //生成签名 用户
    public static String createSigns(String characterEncoding, SortedMap<String, Object> parameters) {
    	StringBuffer sb = new StringBuffer();
    	Set es = parameters.entrySet();
    	Iterator it = es.iterator();
    	while (it.hasNext()) {
    		Map.Entry entry = (Map.Entry) it.next();
    		String k = (String) entry.getKey();
    		Object v = entry.getValue();
    		if (null != v && !"".equals(v)
    				&& !"sign".equals(k) && !"key".equals(k)) {
    			sb.append(k + "=" + v + "&");
    		}
    	}
    	sb.append("key=" + QpayConstants.API_KEY);
    	System.out.println(sb.toString());
    	String sign = MD5Util.MD5Encode(sb.toString(), characterEncoding).toUpperCase();
    	return sign;
    }
    //生成签名
    public static String createSign(String characterEncoding, SortedMap<String, Object> parameters, String key) {
        StringBuffer sb = new StringBuffer();
        Set es = parameters.entrySet();
        Iterator it = es.iterator();
        while (it.hasNext()) {
            Map.Entry entry = (Map.Entry) it.next();
            String k = (String) entry.getKey();
            Object v = entry.getValue();
            if (null != v && !"".equals(v)
                    && !"sign".equals(k) && !"key".equals(k)) {
                sb.append(k + "=" + v + "&");
            }
        }
        //sb.append("key=" + PayConstant.API_KEY);
        sb.append("key=" + key);
        System.out.println(sb.toString());
        String sign = MD5Util.MD5Encode(sb.toString(), characterEncoding).toUpperCase();
        return sign;
    }

    /**
     * 验证回调签名 yonghu
     * @return
     */
    public static boolean isTenpaySigns(Map<String, String> map) {
    	String charset = "utf-8";
    	String signFromAPIResponse = map.get("sign");
    	if (signFromAPIResponse == null || signFromAPIResponse.equals("")) {
    		System.out.println("API返回的数据签名数据不存在，有可能被第三方篡改!!!");
    		return false;
    	}
    	System.out.println("服务器回包里面的签名是:" + signFromAPIResponse);
    	//过滤空 设置 TreeMap
    	SortedMap<String, String> packageParams = new TreeMap<>();
    	for (String parameter : map.keySet()) {
    		String parameterValue = map.get(parameter);
    		String v = "";
    		if (null != parameterValue) {
    			v = parameterValue.trim();
    		}
    		packageParams.put(parameter, v);
    	}
    	
    	StringBuffer sb = new StringBuffer();
    	Set es = packageParams.entrySet();
    	Iterator it = es.iterator();
    	while (it.hasNext()) {
    		Map.Entry entry = (Map.Entry) it.next();
    		String k = (String) entry.getKey();
    		String v = (String) entry.getValue();
    		if (!"sign".equals(k) && null != v && !"".equals(v)) {
    			sb.append(k + "=" + v + "&");
    		}
    	}
    	sb.append("key=" + QpayConstants.API_KEY);
    	//将API返回的数据根据用签名算法进行计算新的签名，用来跟API返回的签名进行比较
    	
    	//算出签名
    	String resultSign = "";
    	String tobesign = sb.toString();
    	
    	if (null == charset || "".equals(charset)) {
    		resultSign = MD5Util.MD5Encode(tobesign, QpayConstants.CHARSET).toUpperCase();
    	} else {
    		resultSign = MD5Util.MD5Encode(tobesign, QpayConstants.CHARSET).toUpperCase();
    	}
    	String tenpaySign = ((String) packageParams.get("sign")).toUpperCase();
    	return tenpaySign.equals(resultSign);
    }
    /**
     * 验证回调签名
     * @return
     */
    public static boolean isTenpaySign(Map<String, String> map,String key) {
        String charset = "utf-8";
        String signFromAPIResponse = map.get("sign");
        if (signFromAPIResponse == null || signFromAPIResponse.equals("")) {
            System.out.println("API返回的数据签名数据不存在，有可能被第三方篡改!!!");
            return false;
        }
        System.out.println("服务器回包里面的签名是:" + signFromAPIResponse);
        System.out.println(map+"微信返回的数据");
        //过滤空 设置 TreeMap
        SortedMap<String, String> packageParams = new TreeMap<>();
        for (String parameter : map.keySet()) {
            String parameterValue = map.get(parameter);
            String v = "";
            if (null != parameterValue) {
                v = parameterValue.trim();
            }
            packageParams.put(parameter, v);
        }

        StringBuffer sb = new StringBuffer();
        Set es = packageParams.entrySet();
        System.out.println(packageParams+"获取微信返回的数据进行对比");
        Iterator it = es.iterator();
        while (it.hasNext()) {
            Map.Entry entry = (Map.Entry) it.next();
            String k = (String) entry.getKey();
            String v = (String) entry.getValue();
            if (!"sign".equals(k) && null != v && !"".equals(v)) {
                sb.append(k + "=" + v + "&");
            }
        }
        sb.append("key=" + key);

        //将API返回的数据根据用签名算法进行计算新的签名，用来跟API返回的签名进行比较

        //算出签名
        String resultSign = "";
        String tobesign = sb.toString();
        System.out.println(tobesign.toString()+"????????????????????????????????");
        if (null == charset || "".equals(charset)) {
            resultSign = MD5Util.MD5Encode(tobesign, QpayConstants.CHARSET).toUpperCase();
        } else {
            resultSign = MD5Util.MD5Encode(tobesign, QpayConstants.CHARSET).toUpperCase();
        }
        System.out.println(resultSign+"再次生成的签名++++++++++++++++++++++++++++++++++++++++++++++++++++");
        String tenpaySign = ((String) packageParams.get("sign")).toUpperCase();
        System.out.println(tenpaySign+"微信返回的签名____________________________________________________");
        return tenpaySign.equals(resultSign);
    }

    //请求方法
    public static String httpsRequest(String requestUrl, String requestMethod, String outputStr) {
        try {

            URL url = new URL(requestUrl);
            HttpURLConnection conn = (HttpURLConnection) url.openConnection();

            conn.setDoOutput(true);
            conn.setDoInput(true);
            conn.setUseCaches(false);
            // 设置请求方式（GET/POST）
            conn.setRequestMethod(requestMethod);
            conn.setRequestProperty("content-type", "application/x-www-form-urlencoded");
            // 当outputStr不为null时向输出流写数据
            if (null != outputStr) {
                OutputStream outputStream = conn.getOutputStream();
                // 注意编码格式
                outputStream.write(outputStr.getBytes("UTF-8"));
                outputStream.close();
            }
            // 从输入流读取返回内容
            InputStream inputStream = conn.getInputStream();
            InputStreamReader inputStreamReader = new InputStreamReader(inputStream, "utf-8");
            BufferedReader bufferedReader = new BufferedReader(inputStreamReader);
            String str = null;
            StringBuffer buffer = new StringBuffer();
            while ((str = bufferedReader.readLine()) != null) {
                buffer.append(str);
            }
            // 释放资源
            bufferedReader.close();
            inputStreamReader.close();
            inputStream.close();
            inputStream = null;
            conn.disconnect();
            return buffer.toString();
        } catch (ConnectException ce) {
            System.out.println("连接超时：{}" + ce);
        } catch (Exception e) {
            System.out.println("https请求异常：{}" + e);
        }
        return null;
    }

    /*
        //退款的请求方法
        public static String httpsRequest2(String requestUrl, String requestMethod, String outputStr) throws Exception {
            KeyStore keyStore  = KeyStore.getInstance("PKCS12");
            StringBuilder res = new StringBuilder("");
            FileInputStream instream = new FileInputStream(new File("/home/apiclient_cert.p12"));
            try {
                keyStore.load(instream, "".toCharArray());
            } finally {
                instream.close();
            }

            // Trust own CA and all self-signed certs
            SSLContext sslcontext = SSLContexts.custom()
                    .loadKeyMaterial(keyStore, "1313329201".toCharArray())
                    .build();
            // Allow TLSv1 protocol only
            SSLConnectionSocketFactory sslsf = new SSLConnectionSocketFactory(
                    sslcontext,
                    new String[] { "TLSv1" },
                    null,
                    SSLConnectionSocketFactory.BROWSER_COMPATIBLE_HOSTNAME_VERIFIER);
            CloseableHttpClient httpclient = HttpClients.custom()
                    .setSSLSocketFactory(sslsf)
                    .build();
            try {

                HttpPost httpost = new HttpPost("https://api.mch.weixin.qq.com/secapi/pay/refund");
                httpost.addHeader("Connection", "keep-alive");
                httpost.addHeader("Accept", "");
                httpost.addHeader("Content-Type", "application/x-www-form-urlencoded; charset=UTF-8");
                httpost.addHeader("Host", "api.mch.weixin.qq.com");
                httpost.addHeader("X-Requested-With", "XMLHttpRequest");
                httpost.addHeader("Cache-Control", "max-age=0");
                httpost.addHeader("User-Agent", "Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 6.0) ");
                StringEntity entity2 = new StringEntity(outputStr ,Consts.UTF_8);
                httpost.setEntity(entity2);
                System.out.println("executing request" + httpost.getRequestLine());

                CloseableHttpResponse response = httpclient.execute(httpost);

                try {
                    HttpEntity entity = response.getEntity();

                    System.out.println("----------------------------------------");
                    System.out.println(response.getStatusLine());
                    if (entity != null) {
                        System.out.println("Response content length: " + entity.getContentLength());
                        BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(entity.getContent()));
                        String text = "";
                        res.append(text);
                        while ((text = bufferedReader.readLine()) != null) {
                            res.append(text);
                            System.out.println(text);
                        }

                    }
                    EntityUtils.consume(entity);
                } finally {
                    response.close();
                }
            } finally {
                httpclient.close();
            }
            return  res.toString();

        }

        */
    //xml解析
    public static Map doXMLParse(String strxml) throws JDOMException, IOException {
        strxml = strxml.replaceFirst("encoding=\".*\"", "encoding=\"UTF-8\"");

        if (null == strxml || "".equals(strxml)) {
            return null;
        }

        Map m = new HashMap();

        InputStream in = new ByteArrayInputStream(strxml.getBytes("UTF-8"));
        SAXBuilder builder = new SAXBuilder();
        Document doc = builder.build(in);
        Element root = doc.getRootElement();
        List list = root.getChildren();
        Iterator it = list.iterator();
        while (it.hasNext()) {
            Element e = (Element) it.next();
            String k = e.getName();
            String v = "";
            List children = e.getChildren();
            if (children.isEmpty()) {
                v = e.getTextNormalize();
            } else {
                v = getChildrenText(children);
            }

            m.put(k, v);
        }

        //关闭流
        in.close();

        return m;
    }

    public static String getChildrenText(List children) {
        StringBuffer sb = new StringBuffer();
        if (!children.isEmpty()) {
            Iterator it = children.iterator();
            while (it.hasNext()) {
                Element e = (Element) it.next();
                String name = e.getName();
                String value = e.getTextNormalize();
                List list = e.getChildren();
                sb.append("<" + name + ">");
                if (!list.isEmpty()) {
                    sb.append(getChildrenText(list));
                }
                sb.append(value);
                sb.append("</" + name + ">");
            }
        }

        return sb.toString();
    }

    //获取真实IP
    public static String getIp(HttpServletRequest request) {
        String ip = request.getHeader("X-Forwarded-For");
        if (!StringUtils.isBlank(ip) && !"unKnown".equalsIgnoreCase(ip)) {
            //多次反向代理后会有多个ip值，第一个ip才是真实ip
            int index = ip.indexOf(",");
            if (index != -1) {
                return ip.substring(0, index);
            } else {
                return ip;
            }
        }
        ip = request.getHeader("X-Real-IP");
        if (!StringUtils.isBlank(ip) && !"unKnown".equalsIgnoreCase(ip)) {
            return ip;
        }
        return request.getRemoteAddr();
    }

    /**
     * 支付成功，返回微信那服务器
     *
     * @param return_code
     * @param return_msg
     * @return
     */
    public static String getXML(String return_code, String return_msg) {
        return "<xml><return_code><![CDATA[" + return_code + "]]></return_code><return_msg><![CDATA[" + return_msg + "]]></return_msg></xml>";
    }


    public static void main(String[] args) {
        String s = MD5Util.MD5Encode("key=wxf0adceacc97514ce", "utf-8").toUpperCase();
        System.out.println(s);
    }



}